# tts_module.py

import websocket
import datetime
import hashlib
import base64
import hmac
import json
from urllib.parse import urlencode
import time
import ssl
from wsgiref.handlers import format_date_time
from datetime import datetime
from time import mktime
import os
import _thread as thread

class TextToSpeech(object):
    def __init__(self, APPID, APIKey, APISecret, Text, output_file):
        self.APPID = APPID
        self.APIKey = APIKey
        self.APISecret = APISecret
        self.Text = Text
        self.output_file = output_file
        self.ws = None

    def create_url(self):
        url = 'wss://tts-api.xfyun.cn/v2/tts'
        now = datetime.now()
        date = format_date_time(mktime(now.timetuple()))

        signature_origin = "host: " + "ws-api.xfyun.cn" + "\n"
        signature_origin += "date: " + date + "\n"
        signature_origin += "GET " + "/v2/tts " + "HTTP/1.1"

        signature_sha = hmac.new(self.APISecret.encode('utf-8'), signature_origin.encode('utf-8'),
                                 digestmod=hashlib.sha256).digest()
        signature_sha = base64.b64encode(signature_sha).decode(encoding='utf-8')

        authorization_origin = "api_key=\"%s\", algorithm=\"%s\", headers=\"%s\", signature=\"%s\"" % (
            self.APIKey, "hmac-sha256", "host date request-line", signature_sha)
        authorization = base64.b64encode(authorization_origin.encode('utf-8')).decode(encoding='utf-8')

        v = {
            "authorization": authorization,
            "date": date,
            "host": "ws-api.xfyun.cn"
        }

        url = url + '?' + urlencode(v)
        return url

    def on_message(self, ws, message):
        try:
            message = json.loads(message)
            code = message["code"]
            sid = message["sid"]
            audio = message["data"]["audio"]
            audio = base64.b64decode(audio)
            status = message["data"]["status"]
            if status == 2:
                print("ws is closed")
                ws.close()
            if code != 0:
                errMsg = message["message"]
                print("sid:%s call error:%s code is:%s" % (sid, errMsg, code))
            else:
                with open(self.output_file, 'ab') as f:
                    f.write(audio)
        except Exception as e:
            print("receive msg, but parse exception:", e)

    def on_error(self, ws, error):
        print("### error:", error)

    def on_close(*args):
        print("### closed ###")

    def on_open(self, ws):
        def run(*args):
            d = {"common": {"app_id": self.APPID},
                 "business": {"aue": "raw", "auf": "audio/L16;rate=8000", "vcn": "xiaoyan", "tte": "utf8"},
                 "data": {"status": 2, "text": str(base64.b64encode(self.Text.encode('utf-8')), "UTF8")}
                 }
            d = json.dumps(d)
            print("------>开始发送文本数据")
            ws.send(d)
            if os.path.exists(self.output_file):
                os.remove(self.output_file)

        thread.start_new_thread(run, ())

    def tts(self):
        websocket.enableTrace(False)
        url = self.create_url()
        self.ws = websocket.WebSocketApp(url, on_message=self.on_message, on_error=self.on_error, on_close=self.on_close)
        self.ws.on_open = self.on_open
        self.ws.run_forever(sslopt={"cert_reqs": ssl.CERT_NONE})

if __name__ == "__main__":
    APPID = '21fa1ed5'
    APISecret = 'YzFlNmFiMjJkYTIwMTFhNWZkYjg4NDlm'
    APIKey = 'b23ea50fd4cbaf35a9a502c47cb8cd4c'
    text = "这是一个语音合成示例"
    output_file = "output.pcm"
    tts = TextToSpeech(APPID, APIKey, APISecret, text, output_file)
    tts.tts()
# # -*- coding:utf-8 -*-
# #
# #   author: iflytek
# #
# #  本demo测试时运行的环境为：Windows + Python3.7
# #  本demo测试成功运行时所安装的第三方库及其版本如下：
# #   cffi==1.12.3
# #   gevent==1.4.0
# #   greenlet==0.4.15
# #   pycparser==2.19
# #   six==1.12.0
# #   websocket==0.2.1
# #   websocket-client==0.56.0
# #   合成小语种需要传输小语种文本、使用小语种发音人vcn、tte=unicode以及修改文本编码方式
# #  错误码链接：https://www.xfyun.cn/document/error-code （code返回错误码时必看）
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# import websocket
# import datetime
# import hashlib
# import base64
# import hmac
# import json
# from urllib.parse import urlencode
# import time
# import ssl
# from wsgiref.handlers import format_date_time
# from datetime import datetime
# from time import mktime
# import _thread as thread
# import os
# import pyaudio
# import numpy as np

# STATUS_FIRST_FRAME = 0  # 第一帧的标识
# STATUS_CONTINUE_FRAME = 1  # 中间帧标识
# STATUS_LAST_FRAME = 2  # 最后一帧的标识


# # 配置参数
# FORMAT = pyaudio.paInt16
# CHANNELS = 1
# RATE = 16000
# CHUNK = 256

# # 初始化音频流
# p = pyaudio.PyAudio()
# stream = p.open(format=FORMAT,
#                 channels=CHANNELS,
#                 rate=RATE,
#                 output=True,
#                 frames_per_buffer=CHUNK)  


# class Speech_tts(object):
#     # 初始化
#     def __init__(self, APPID, APIKey, APISecret, Text):
#         self.APPID = APPID
#         self.APIKey = APIKey
#         self.APISecret = APISecret
#         self.Text = Text
#         self.audio_file = None
#         self.out_audio_data = b''
#         self.ws = None
#         # 公共参数(common)
#         self.CommonArgs = {"app_id": self.APPID}
#         # 业务参数(business)，更多个性化参数可在官网查看
#         #self.BusinessArgs = {"aue": "speex-org-nb;8", "auf": "audio/L16;rate=8000", "vcn": "x4_lingxiaoyao_em", "tte": "utf8"}
#         #self.BusinessArgs = {"aue":"raw", "auf": "audio/L16;rate=16000", "vcn": "x4_lingxiaoyao_em", "tte": "utf8"}
#         self.BusinessArgs = {"aue": "raw", "auf": "audio/L16;rate=16000", "vcn": "xiaoyan", "tte": "utf8"}
#         self.Data = {"status": 2, "text": str(base64.b64encode(self.Text.encode('utf-8')), "UTF8")}
#         #使用小语种须使用以下方式，此处的unicode指的是 utf16小端的编码方式，即"UTF-16LE"”
#         #self.Data = {"status": 2, "text": str(base64.b64encode(self.Text.encode('utf-16')), "UTF8")}

#     # 生成url
#     def create_url(self):
#         url = 'wss://tts-api.xfyun.cn/v2/tts'
#         # 生成RFC1123格式的时间戳
#         now = datetime.now()
#         date = format_date_time(mktime(now.timetuple()))

#         # 拼接字符串
#         signature_origin = "host: " + "ws-api.xfyun.cn" + "\n"
#         signature_origin += "date: " + date + "\n"
#         signature_origin += "GET " + "/v2/tts " + "HTTP/1.1"
#         # 进行hmac-sha256进行加密
#         signature_sha = hmac.new(self.APISecret.encode('utf-8'), signature_origin.encode('utf-8'),
#                                  digestmod=hashlib.sha256).digest()
#         signature_sha = base64.b64encode(signature_sha).decode(encoding='utf-8')

#         authorization_origin = "api_key=\"%s\", algorithm=\"%s\", headers=\"%s\", signature=\"%s\"" % (
#             self.APIKey, "hmac-sha256", "host date request-line", signature_sha)
#         authorization = base64.b64encode(authorization_origin.encode('utf-8')).decode(encoding='utf-8')
#         # 将请求的鉴权参数组合为字典
#         v = {
#             "authorization": authorization,
#             "date": date,
#             "host": "ws-api.xfyun.cn"
#         }
#         # 拼接鉴权参数，生成url
#         url = url + '?' + urlencode(v)
#         # print("date: ",date)
#         # print("v: ",v)
#         # 此处打印出建立连接时候的url,参考本demo的时候可取消上方打印的注释，比对相同参数时生成的url与自己代码生成的url是否一致
#         # print('websocket url :', url)
#         return url

#     def run(self, audio_file):
#         self.audio_file = audio_file
#         websocket.enableTrace(False)
#         url = self.create_url()
#         self.ws = websocket.WebSocketApp(url, on_message=self.on_message, on_error=self.on_error, on_close=self.on_close)
#         self.ws.on_open = self.on_open
#         self.ws.run_forever(sslopt={"cert_reqs": ssl.CERT_NONE})



#     def get_audio_data(self, audio_file):
#     # 如果之前有 WebSocket 连接，则先关闭
#         if self.ws is not None:
#             self.ws.close()
#         #self.out_audio_data = b''
#         self.run(audio_file)
#         #return self.out_audio_data

    
#     def on_message(self,ws, message):
#         try:
#             message =json.loads(message)
#             code = message["code"]
#             sid = message["sid"]
#             audio = message["data"]["audio"]
#             audio = base64.b64decode(audio)
#             status = message["data"]["status"]
#             print(message)
#             if status == 2:
#                 print("ws is closed")
#                 ws.close()
#             if code != 0:
#                 errMsg = message["message"]
#                 print("sid:%s call error:%s code is:%s" % (sid, errMsg, code))
#             else:
#                 #self.out_audio_data+=audio
#                 with open(self.audio_file, 'ab') as f:
#                     f.write(audio)
#                     #stream.write(audio)
#         except Exception as e:
#             print("receive msg,but parse exception:", e)



#     # 收到websocket错误的处理
#     def on_error(ws,error,srg):
#         print("### error:",error)


#     # 收到websocket关闭的处理
#     def on_close(ws,error,srg):
#         print("### closed ###")


#     # 收到websocket连接建立的处理
#     def on_open(self, ws):
#         def run(*args):
#             d = {"common": self.CommonArgs,
#                 "business": self.BusinessArgs,
#                 "data": self.Data,
#                 }
#             d = json.dumps(d)
#             print("------>开始发送文本数据")
#             ws.send(d)
#             if os.path.exists(self.audio_file):
#                 os.remove(self.audio_file)

#         thread.start_new_thread(run, ())









# if __name__ == "__main__":
#     # 测试时候在此处正确填写相关信息即可运行
#     tts = Speech_tts(APPID='21fa1ed5', APISecret='YzFlNmFiMjJkYTIwMTFhNWZkYjg4NDlm',
#                        APIKey='b23ea50fd4cbaf35a9a502c47cb8cd4c',
#                        Text="这是一个语音合成示例")
#     tts.get_audio_data("tts.speex")
#     # websocket.enableTrace(False)
#     # wsUrl = tts.create_url()
#     # ws = websocket.WebSocketApp(wsUrl, on_message=on_message, on_error=on_error, on_close=on_close)
#     # ws.on_open = on_open
#     # ws.run_forever(sslopt={"cert_reqs": ssl.CERT_NONE})
